home *** CD-ROM | disk | FTP | other *** search
/ PC World 2006 February / PCWorld_2006-02_cd.bin / software / vyzkuste / triky / triky.exe / httrack-3.33.exe / {app} / src / httrack.h < prev    next >
C/C++ Source or Header  |  2004-10-05  |  9KB  |  299 lines
  1. /* ------------------------------------------------------------ */
  2. /*
  3. HTTrack Website Copier, Offline Browser for Windows and Unix
  4. Copyright (C) Xavier Roche and other contributors
  5.  
  6. This program is free software; you can redistribute it and/or
  7. modify it under the terms of the GNU General Public License
  8. as published by the Free Software Foundation; either version 2
  9. of the License, or any later version.
  10.  
  11. This program is distributed in the hope that it will be useful,
  12. but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14. GNU General Public License for more details.
  15.  
  16. You should have received a copy of the GNU General Public License
  17. along with this program; if not, write to the Free Software
  18. Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  19.  
  20.  
  21. Important notes:
  22.  
  23. - We hereby ask people using this source NOT to use it in purpose of grabbing
  24. emails addresses, or collecting any other private information on persons.
  25. This would disgrace our work, and spoil the many hours we spent on it.
  26.  
  27.  
  28. Please visit our Website: http://www.httrack.com
  29. */
  30.  
  31.  
  32. /* ------------------------------------------------------------ */
  33. /* File: htsshow.c console progress info                        */
  34. /* Author: Xavier Roche                                         */
  35. /* ------------------------------------------------------------ */
  36.  
  37. #ifndef HTSTOOLS_DEFH
  38. #define HTSTOOLS_DEFH 
  39.  
  40. #if HTS_ANALYSTE_CONSOLE
  41.  
  42. #include "htsglobal.h"
  43. #include "htscore.h"
  44.  
  45. typedef struct {
  46.   char name[1024];
  47.   char file[1024];
  48.   char state[256];
  49.   char BIGSTK url_sav[HTS_URLMAXSIZE*2];    // pour cancel
  50.   char BIGSTK url_adr[HTS_URLMAXSIZE*2];
  51.   char BIGSTK url_fil[HTS_URLMAXSIZE*2];
  52.   LLint size;
  53.   LLint sizetot;
  54.   int offset;
  55.   //
  56.   int back;
  57.   //
  58.   int actived;    // pour disabled
  59. } t_StatsBuffer;
  60.  
  61. typedef struct {
  62.   int ask_refresh;
  63.   int refresh;
  64.   LLint stat_bytes;
  65.   int stat_time;
  66.   int lien_n;
  67.   int lien_tot;
  68.   int stat_nsocket;
  69.   int rate;
  70.   int irate;
  71.   int ft;
  72.   LLint stat_written;
  73.   int stat_updated;
  74.   int stat_errors;
  75.   int stat_warnings;
  76.   int stat_infos;
  77.   TStamp stat_timestart;
  78.   int stat_back;
  79. } t_InpInfo;
  80.  
  81. // wrappers
  82. static void  __cdecl htsshow_init(void);
  83. static void  __cdecl htsshow_uninit(void);
  84. static int   __cdecl htsshow_start(httrackp* opt);
  85. static int   __cdecl htsshow_chopt(httrackp* opt);
  86. static int   __cdecl htsshow_end(void);
  87. static int   __cdecl htsshow_preprocesshtml(char** html,int* len,char* url_adresse,char* url_fichier);
  88. static int   __cdecl htsshow_postprocesshtml(char** html,int* len,char* url_adresse,char* url_fichier);
  89. static int   __cdecl htsshow_checkhtml(char* html,int len,char* url_adresse,char* url_fichier);
  90. static int   __cdecl htsshow_loop(lien_back* back,int back_max,int back_index,int lien_n,int lien_tot,int stat_time,hts_stat_struct* stats);
  91. static char* __cdecl htsshow_query(char* question);
  92. static char* __cdecl htsshow_query2(char* question);
  93. static char* __cdecl htsshow_query3(char* question);
  94. static int   __cdecl htsshow_check(char* adr,char* fil,int status);
  95. static void  __cdecl htsshow_pause(char* lockfile);
  96. static void  __cdecl htsshow_filesave(char* file);
  97. static int   __cdecl htsshow_linkdetected(char* link);
  98. static int   __cdecl htsshow_linkdetected2(char* link, char* start_tag);
  99. static int   __cdecl htsshow_xfrstatus(lien_back* back);
  100. static int   __cdecl htsshow_savename(char* adr_complete,char* fil_complete,char* referer_adr,char* referer_fil,char* save);
  101. static int   __cdecl htsshow_sendheader(char* buff, char* adr, char* fil, char* referer_adr, char* referer_fil, htsblk* outgoing);
  102. static int   __cdecl htsshow_receiveheader(char* buff, char* adr, char* fil, char* referer_adr, char* referer_fil, htsblk* incoming);
  103.  
  104. int main(int argc, char **argv);
  105. static void vt_color(int text,int back);
  106. static void vt_clear(void);
  107. static void vt_home(void);
  108.  
  109. #endif
  110.  
  111. /* */
  112.  
  113. // Engine internal variables
  114. typedef void (* htsErrorCallback)(char* msg, char* file, int line);
  115. extern HTSEXT_API htsErrorCallback htsCallbackErr;
  116. extern HTSEXT_API int htsMemoryFastXfr;
  117. /* */
  118. extern HTSEXT_API hts_stat_struct HTS_STAT;
  119. extern int _DEBUG_HEAD;
  120. extern FILE* ioinfo;
  121.  
  122. // from htsbase.h
  123.  
  124. /* protected strcat, strncat and strcpy - definitely useful */
  125. #define strcatbuff(A, B) do { \
  126.   assertf( (A) != NULL ); \
  127.   if ( ! (B) ) { assertf( 0 ); } \
  128.   if (htsMemoryFastXfr) { \
  129.     if (sizeof(A) != sizeof(char*)) { \
  130.       (A)[sizeof(A) - 1] = '\0'; \
  131.     } \
  132.     strcat(A, B); \
  133.     if (sizeof(A) != sizeof(char*)) { \
  134.       assertf((A)[sizeof(A) - 1] == '\0'); \
  135.     } \
  136.   } else { \
  137.     unsigned int sz = (unsigned int) strlen(A); \
  138.     unsigned int szf = (unsigned int) strlen(B); \
  139.     if (sizeof(A) != sizeof(char*)) { \
  140.       assertf(sz + szf + 1 < sizeof(A)); \
  141.       if (szf > 0) { \
  142.         if (sz + szf + 1 < sizeof(A)) { \
  143.           memcpy((A) + sz, (B), szf + 1); \
  144.         } \
  145.       } \
  146.     } else if (szf > 0) { \
  147.       memcpybuff((A) + sz, (B), szf + 1); \
  148.     } \
  149.   } \
  150. } while(0)
  151. #define strncatbuff(A, B, N) do { \
  152.   assertf( (A) != NULL ); \
  153.   if ( ! (B) ) { assertf( 0 ); } \
  154.   if (htsMemoryFastXfr) { \
  155.     if (sizeof(A) != sizeof(char*)) { \
  156.       (A)[sizeof(A) - 1] = '\0'; \
  157.     } \
  158.     strncat(A, B, N); \
  159.     if (sizeof(A) != sizeof(char*)) { \
  160.       assertf((A)[sizeof(A) - 1] == '\0'); \
  161.     } \
  162.   } else { \
  163.     unsigned int sz = (unsigned int) strlen(A); \
  164.     unsigned int szf = (unsigned int) strlen(B); \
  165.     if (szf > (unsigned int) (N)) szf = (unsigned int) (N); \
  166.     if (sizeof(A) != sizeof(char*)) { \
  167.       assertf(sz + szf + 1 < sizeof(A)); \
  168.       if (szf > 0) { \
  169.         if (sz + szf + 1 < sizeof(A)) { \
  170.           memcpy((A) + sz, (B), szf); \
  171.           * ( (A) + sz + szf) = '\0'; \
  172.         } \
  173.       } \
  174.     } else if (szf > 0) { \
  175.       memcpybuff((A) + sz, (B), szf); \
  176.       * ( (A) + sz + szf) = '\0'; \
  177.     } \
  178.   } \
  179. } while(0)
  180. #define strcpybuff(A, B) do { \
  181.   assertf( (A) != NULL ); \
  182.   if ( ! (B) ) { assertf( 0 ); } \
  183.   if (htsMemoryFastXfr) { \
  184.     if (sizeof(A) != sizeof(char*)) { \
  185.       (A)[sizeof(A) - 1] = '\0'; \
  186.     } \
  187.     strcpy(A, B); \
  188.     if (sizeof(A) != sizeof(char*)) { \
  189.       assertf((A)[sizeof(A) - 1] == '\0'); \
  190.     } \
  191.   } else { \
  192.     unsigned int szf = (unsigned int) strlen(B); \
  193.     if (sizeof(A) != sizeof(char*)) { \
  194.       assertf(szf + 1 < sizeof(A)); \
  195.       if (szf > 0) { \
  196.         if (szf + 1 < sizeof(A)) { \
  197.           memcpy((A), (B), szf + 1); \
  198.         } else { \
  199.           * (A) = '\0'; \
  200.         } \
  201.       } else { \
  202.         * (A) = '\0'; \
  203.       } \
  204.     } else { \
  205.       memcpybuff((A), (B), szf + 1); \
  206.     } \
  207.   } \
  208. } while(0)
  209. #define strncpybuff(A, B, N) do { \
  210.   assertf( (A) != NULL ); \
  211.   if ( ! (B) ) { assertf( 0 ); } \
  212.   if (htsMemoryFastXfr) { \
  213.     if (sizeof(A) != sizeof(char*)) { \
  214.       (A)[sizeof(A) - 1] = '\0'; \
  215.     } \
  216.     strncpy(A, B, N); \
  217.     if (sizeof(A) != sizeof(char*)) { \
  218.       assertf((A)[sizeof(A) - 1] == '\0'); \
  219.     } \
  220.   } else { \
  221.     unsigned int szf = (unsigned int) strlen(B); \
  222.     if (szf > (unsigned int) (N)) szf = (unsigned int) (N); \
  223.     if (sizeof(A) != sizeof(char*)) { \
  224.       assertf(szf + 1 < sizeof(A)); \
  225.       if (szf > 0) { \
  226.         if (szf + 1 < sizeof(A)) { \
  227.           memcpy((A), (B), szf); \
  228.         } \
  229.       } \
  230.     } else { \
  231.       memcpybuff((A), (B), szf); \
  232.     } \
  233.   } \
  234. } while(0)
  235.  
  236. // emergency log
  237. typedef void (*t_abortLog)(char* msg, char* file, int line);
  238. extern HTSEXT_API t_abortLog abortLog__;
  239. #define abortLog(a) abortLog__(a, __FILE__, __LINE__)
  240. #define abortLogFmt(a) do { \
  241.   FILE* fp = fopen("CRASH.TXT", "wb"); \
  242.   if (!fp) fp = fopen("/tmp/CRASH.TXT", "wb"); \
  243.   if (!fp) fp = fopen("C:\\CRASH.TXT", "wb"); \
  244.   if (fp) { \
  245.     fprintf(fp, "HTTrack " HTTRACK_VERSIONID " closed at '" __FILE__ "', line %d\r\n", __LINE__); \
  246.     fprintf(fp, "Reason:\r\n"); \
  247.     fprintf(fp, a); \
  248.     fprintf(fp, "\r\n"); \
  249.     fflush(fp); \
  250.     fclose(fp); \
  251.   } \
  252. } while(0)
  253.  
  254. #define _ ,
  255. #define abortLogFmt(a) do { \
  256.   FILE* fp = fopen("CRASH.TXT", "wb"); \
  257.   if (!fp) fp = fopen("/tmp/CRASH.TXT", "wb"); \
  258.   if (!fp) fp = fopen("C:\\CRASH.TXT", "wb"); \
  259.   if (fp) { \
  260.     fprintf(fp, "HTTrack " HTTRACK_VERSIONID " closed at '" __FILE__ "', line %d\r\n", __LINE__); \
  261.     fprintf(fp, "Reason:\r\n"); \
  262.     fprintf(fp, a); \
  263.     fprintf(fp, "\r\n"); \
  264.     fflush(fp); \
  265.     fclose(fp); \
  266.   } \
  267. } while(0)
  268. #define assertf(exp) do { \
  269.   if (! ( exp ) ) { \
  270.     abortLog("assert failed: " #exp); \
  271.     if (htsCallbackErr != NULL) { \
  272.       htsCallbackErr("assert failed: " #exp, __FILE__ , __LINE__ ); \
  273.     } \
  274.     assert(exp); \
  275.     abort(); \
  276.   } \
  277. } while(0)
  278. /* non-fatal assert */
  279. #define assertnf(exp) do { \
  280.   if (! ( exp ) ) { \
  281.     abortLog("assert failed: " #exp); \
  282.     if (htsCallbackErr != NULL) { \
  283.       htsCallbackErr("assert failed: " #exp, __FILE__ , __LINE__ ); \
  284.     } \
  285.   } \
  286. } while(0)
  287.  
  288. //
  289.  
  290. #define malloct(A)          malloc(A)
  291. #define calloct(A,B)        calloc((A), (B))
  292. #define freet(A)            do { assertnf((A) != NULL); if ((A) != NULL) { free(A); (A) = NULL; } } while(0)
  293. #define strdupt(A)          strdup(A)
  294. #define realloct(A,B)       ( ((A) != NULL) ? realloc((A), (B)) : malloc(B) )
  295. #define memcpybuff(A, B, N) memcpy((A), (B), (N))
  296.  
  297. #endif
  298.  
  299.